【Leetcode】【python】Path Sum II 路径总和 II

题目大意

将根到叶子的路径和为sum的路径都枚举出来。

解题思路

递归,并且用了python函数嵌套,有关函数嵌套可以看这一篇文章

其实一开始不想项标准答案一样用函数嵌套,毕竟别的语言可能不支持,以后看答案不方便,但是如果把list_all放在全局,需要每轮都去清空它,而leetcode跑测试的时候应该是一个类的对象跑完所有测试,所以全局变量会累加,只能按照标准答案写了。

注意:

  • 嵌套的函数不需要再写self

  • 内层函数可以访问外层函数中定义的变量,但不能重新赋值(rebind)

    代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution(object):

def pathSum(self, root, sum):
"""
:type root: TreeNode
:type sum: int
:rtype: List[List[int]]
"""
def dfs(root, cur_sum, list_temp): # 不需要self
if root.left == None and root.right == None:
if cur_sum == sum:
return list_all.append(list_temp)
if root.left:
dfs(root.left, cur_sum + root.left.val, list_temp + [root.left.val])
if root.right:
dfs(root.right, cur_sum + root.right.val, list_temp + [root.right.val])
if root == None:
return []
list_all = []
dfs(root, root.val, [root.val])
return list_all

总结